package com.rpframework.core.utils;

import freemarker.template.Configuration;
import freemarker.template.Template;
import freemarker.template.TemplateException;

import java.io.*;
import java.util.Map;

public class FreemarkerUtil {

    public static Template getTemplate(String name, String path) {
        try {
            // 通过Freemaker的Configuration读取相应的ftl
            Configuration cfg = new Configuration();
            // 设定去哪里读取相应的ftl模板文件
            cfg.setClassForTemplateLoading(FreemarkerUtil.class, path);
            // 在模板文件目录中找到名称为name的文件
            Template temp = cfg.getTemplate(name);
            return temp;
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    /**
     * 输出HTML文件
     *
     * @param name
     * @param root
     */
    public static File getFile(String name, Object root, File file, String path) {
        FileWriter out = null;
        try {
            // 通过一个文件输出流，就可以写到相应的文件中，此处用的是绝对路径
            out = new FileWriter(file);
            Template temp = getTemplate(name, path);
            temp.process(root, out);
        } catch (IOException | TemplateException e) {
            throw new RuntimeException(e);
        } finally {
            try {
                if (out != null)
                    out.close();
            } catch (IOException e) {
                throw new RuntimeException(e);
            }
        }
        return file;
    }

    /**
     * 输出HTML文件
     *
     * @param name
     * @param root
     */
    public static File getFile(String name, Object root, File file) {
        return getFile(name, root, file, "/");
    }


    /**
     * 输出HTML文件
     *
     * @param name
     * @param root
     */
    public static String getFile(String path, String name, Map<String, Object> root) {
        return file2String(getFile(name, root, new File(System.getProperty("user.dir") + "0.ftl"), path));
    }

    /**
     * 文本文件转换为指定编码的字符串
     *
     * @param file     文本文件
     * @param encoding 编码类型
     * @return 转换后的字符串
     * @throws IOException
     */
    public static String file2String(File file, String encoding) {
        InputStreamReader reader = null;
        StringWriter writer = new StringWriter();
        try {
            if (encoding == null || "".equals(encoding.trim())) {
                reader = new InputStreamReader(new FileInputStream(file), encoding);
            } else {
                reader = new InputStreamReader(new FileInputStream(file));
            }
            // 将输入流写入输出流
            char[] buffer = new char[4096];
            int n = 0;
            while (-1 != (n = reader.read(buffer))) {
                writer.write(buffer, 0, n);
            }
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        } finally {
            if (reader != null)
                try {
                    reader.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
        }
        return writer.toString();
    }

    /**
     * 文本文件转换为指定编码的字符串
     *
     * @param file 文本文件
     * @return 转换后的字符串
     * @throws IOException
     */
    public static String file2String(File file) {
        return file2String(file, "utf-8");
    }
}